package com.yuandengta.yomall.seckill.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

/**
 * Rabbit 消息队列配置 {@link RabbitConfig}
 *
 * @author Kevin
 * @email: 178676392@qq.com
 */
@Slf4j
@Configuration
public class RabbitConfig {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * 使用JSON序列化机制进行消息转换
     * @return
     */
    @Bean
    public MessageConverter messageConverter(){
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制RabbitTemplate
     * 1.服务器收到消息就回调
     *  - spring.rabbitmq.publisher-confirm-type=correlated
     *  - 设置确认回调ConfirmCallback
     * 2.消息正确抵达队列进行回调
     *  - spring.rabbitmq.publisher-returns=true
     *  - spring.rabbitmq.template.mandatory=true
     *  - 设置确认回调ReturnCallback
     * 3.消费端确认
     *  - spring.rabbitmq.listener.simple.acknowledge-mode=manual 设置手动签收
     *  - channel.basicAck() 签收：业务成功完成就签收
     *  - channel.basicNack() 拒签：业务失败就拒签
     *  - channel.basicReject()
     */
    @PostConstruct
    public void initRabbitTemplate(){
        // 设置确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 只要消息抵达Broker，ack=true
             * @param correlationData 当前消息的唯一关联数据（消息唯一id）
             * @param ack 消息是否成功收到
             * @param cause 失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                // 1.做好消息确认机制
                // 2.每一个发送的消息都在数据库记录，定期将失败的消息再次发送
                // 3.修改消息表消息状态
                log.info("confirm...correlationData:{}, ack:{}, case:{}", correlationData, ack, cause);
            }
        });

        // 设置消息抵达队列的确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要消息没有投递给指定的队列，就触发这个失败回调
             * @param message 投递失败的消息详细信息
             * @param replyCode 回复的状态码
             * @param replyText 回复的文本内容
             * @param exchange 当时这个消息发给哪个交换机
             * @param routingKey 当时这个消息用哪个路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                // 报错，修改消息表当前消息的状态
                log.error("Fail Message:{}, replayCode:{}, replayText:{}, exchange:{}, routingKey:{}",
                        message, replyCode, replyText, exchange, routingKey);
            }
        });
    }
}
